package com.example.util;

import com.example.pojo.mysql.SysDept;
import com.example.pojo.mysql.TreeSelect;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.stream.Collectors;

/* ━━━━━━佛祖保佑━━━━━━
 *                  ,;,,;
 *                ,;;'(    社
 *      __      ,;;' ' \   会
 *   /'  '\'~~'~' \ /'\.)  主
 * ,;(      )    /  |.     义
 *,;' \    /-.,,(   ) \    码
 *     ) /       ) / )|    农
 *     ||        ||  \)
 *     (_\       (_\
 * ━━━━━━永无BUG━━━━━━
 * @author ：zjq
 * @date ：2020/12/10 14:58
 * @description: TODO
 * @version: V1.0
 * @slogan: 天下风云出我辈，一入代码岁月催
 */
public class TreeUtil {


    /**
     * 构建 部门 树
     *
     * @param deptsList
     * @return
     */
    public static List<TreeSelect> buildDeptTreeSelect(List<SysDept> deptsList) {

        List<SysDept> sysDepts = buildDeptTree(deptsList);

        System.out.println("=========================================");
        sysDepts.stream().forEach(System.out::println);
        System.out.println("=========================================");

        List<TreeSelect> collect = sysDepts.stream().map(TreeSelect::new).collect(Collectors.toList());

        return collect;
    }


    public static List<SysDept> buildDeptTree(List<SysDept> depts) {

        List<SysDept> returnList = new ArrayList();


        depts.stream().forEach(n -> {
            // 顶级 节点 遍历所有子节点
            if (n.getParentId() == 0L) {
                recursionFn(depts, n);
                returnList.add(n);
            }
        });


        return returnList;
    }


    /**
     * 递归列表
     */
    private static void recursionFn(List<SysDept> list, SysDept t) {
        // 得到子节点列表
        List<SysDept> childList = getChildList(list, t);
        t.setChildren(childList);
        childList.stream().forEach(n -> {
            if (hasChild(list, n)) {
                recursionFn(list, n);
            }

        });
    }

    /**
     * 得到子节点列表
     */
    private static List<SysDept> getChildList(List<SysDept> list, SysDept t) {
        List<SysDept> tmpList = new ArrayList();
        list.stream().forEach(n -> {
            if (n.getParentId() != null && n.getParentId().longValue() == t.getDeptId().longValue()) {
                // 加入临时list
                tmpList.add(n);
            }
        });
        return tmpList;
    }

    /**
     * 判断是否有子节点
     */
    private static boolean hasChild(List<SysDept> list, SysDept t) {
        return getChildList(list, t).size() > 0 ? true : false;
    }

}
